列表推导式
列表解析是用一行代码实现根据一个列表生成另一个列表的写法。
基本写法
要实现从一个列表中获取满足条件的某些元素组成新的列表:
[x for x in [1,2,3,4] if x % 2 == 0]
这行代码输出是 [2, 4]
。如果没有 if 的话就是直接复制整个列表。
其他用法
修改新列表的元素:
[x/2 for x in [1,2,3,4]]
将单个元素的列表重复此元素,扩展为指定长度的列表:
[x for i in range(100)]
提取嵌套列表中元素的第一项:
[x[0] for x in [[1,2], [3,4], [5,6]]]
嵌套循环的写法:
[n for row in [[1,2], [3,4], [5,6]] for n in row]
生成嵌套列表:
[(x,x+10) for x in [1,2]
多个参数:[(x,y) for x,y in {1:2, 4:5}.items() if x % 2 == 0]
与条件表达式结合:[x if x % 3 == 0 else 1 for x in range(10)]
[(x,y) if x % 2 == 0 else (y, 3) for x,y in {1:2, 4:5}.items() ]
字典推导式
与列表推导式类似,字典也有相应的推导式写法。
基本写法
{x:y for y,x in {'1':'2', '3':'4'}.items() if '1' in y}
参考资料: